import numpy as np

batch_size = 10
a = np.array([i for i in range(100)])
result = np.sum(10 == a[1:1+batch_size])
print(result)
